# model_config
input_shape = (64, 64, 64, 1)
base_nb_filters = 32
weight_decay = 1e-4
center_lr = 0.2
center_weight = 0.5
malign_weight = 1

# train config
lr = 1e-4
seed = 0
batch_size = 16
max_epoch = 200
sample_per_epoch = 4000
